from pathlib import Path

from ipyniivue import download_dataset

BASE_API_URL = "https://niivue.com/demos/images/"
DATA_FOLDER = Path("images")

# Download data for example
download_dataset(
    BASE_API_URL,
    DATA_FOLDER,
    files=[
        "mni152.nii.gz",
        "hippo.nii.gz",
    ],
)

mni152.nii.gz already exists.
hippo.nii.gz already exists.
Dataset downloaded successfully to images.

from ipyniivue import NiiVue, SliceType

volumes = [
    {
        "path": DATA_FOLDER / "mni152.nii.gz",
        "colormap": "gray",
        "opacity": 1.0,
    },
    {
        "path": DATA_FOLDER / "hippo.nii.gz",
        "colormap": "red",
        "opacity": 1,
    },
]
nv = NiiVue(slice_type=SliceType.MULTIPLANAR)


@nv.on_image_loaded
def save_documents(img):
    """Save documents demo."""
    if img.name.endswith("hippo.nii.gz"):
        nv.save_document("test.nvd")
        nv.save_html("test.html")
        nv.save_image("test.nii.gz")
        nv.save_scene("test.png")


nv.load_volumes(volumes)
nv